Project 3 - Waves and Fourier Analysis

Kyle Taft

Background:

The Fourier transform is an exceptional mathematical tool that is useful is many fields of physics. It is used to transform a function from the time domain to the frequency domain. In my project I decided to use it in a more complicated way than we have seen in class. I used it change between position and momentum space for a quantum mechanical wave function. In this notebook I show its importance in being able to numerically solve the Schrodinger equation for a particle in an infinite potenial well with a barrier in the middle by use of FFTs. I produce animations to help visualize the physics of the problem.

Note: This project stands as a proof of concept and many unphysical short cuts were taken for sake of time. A lot of constant are incorrect and the wave function is not always normalized. I used the units that made the animation look the best.

I took motivation from the following project: https://jakevdp.github.io/blog/2012/09/05/quantum-python/

Physics Motivation:

In order to build the mathematic basis of the quantum mechanics and therefore the usefulness of the Fourier transform in this problem I will be walking through equations pulled directly out of McIntyre's Quantum Mechanics textbook.

The fundamental relations that we rely on to make this calculation are equation 6.30 and 6.31 from McIntyre. They are as follows:

$$\psi(x) = \frac{1}{\sqrt{2\pi\hbar}} \int_{-\infty}^{\infty} \phi(p) e^{\frac{ipx}{\hbar}} dp = Fourier(\phi(p))$$$$\phi(p) = \frac{1}{\sqrt{2\pi\hbar}} \int_{-\infty}^{\infty} \psi(x) e^{\frac{ipx}{\hbar}} dx = invFourier(\psi(x))$$

Where $\psi(x)$ is the wave function in position space, $\phi(p)$ is the wave function in momentum space. We are able to figure out one if we know the other through the use of Fourier transforms. We will see how this allows us to do our calculations in our code.

Defining our inital wave function:

Here we generate an example position wave packet that it centered at 0. We can see its shape in the position space in the first plot with its imagniary part. By taking the magnitude squared of the wave packet we can get the probability density of the position.

Defining the potential well:

Here we define the potential well that we will be using for our calculations. We will be using a potential well that is infinite on the left and right side and has a barrier in the middle. Here it is graphed:

Evolving the wave function in time:

Here is the real meat of the project. We will be using the Fourier transform to evolve the wave function in time using the "split-operator" method. The focus is in the function evolve defined below. The function takes in the wave function in position space and evolves it in time. It does this by first taking the Fourier transform of the wave function in position space to get the wave function in momentum space. Then it multiplies the wave function in momentum space by the kinetic energy operator in momentum space. Then it takes the inverse Fourier transform of the wave function in momentum space to get the wave function in position space. Then it multiplies the wave function in position space by the potential energy operator in position space. Then it multiplies the wave function in position space by the time evolution operator. Then it repeats the process for the next time step using the new wave function in position space.

Conclusion:

Again, there are multiple imperfections in the way that the code has been implemented and many jumps over reasoning. Yet, our focus is still clear. We see in our final result the power that the FFT can give us to analyze physical situations. We are able to see the wave function evolve in time and see the effects of the potential well on the wave function. With more time and a closer attention to the physics this technique could be used to solve many more complicated problems that are not easily solvable analytically.